CUDA_VISIBLE_DEVICES=5 python src/extract_embedding.py \
--model mifst \
--pdb_path /home/user4/data/swiss_prot_pdb/ \
--output_path data/swissprot_mifst \
--embed_type mean \
--chunck_id 5